1   /*
2    * Copyright (C) 2008 The Guava Authors
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    * http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package com.google.common.collect.testing;
18  
19  import com.google.common.annotations.GwtCompatible;
20  
21  import java.util.List;
22  import java.util.Map;
23  import java.util.Set;
24  
25  /**
26   * Creates map entries using sample keys and sample values.
27   *
28   * @author Jesse Wilson
29   */
30  @GwtCompatible
31  public abstract class TestMapEntrySetGenerator<K, V>
32      implements TestSetGenerator<Map.Entry<K, V>> {
33    private final SampleElements<K> keys;
34    private final SampleElements<V> values;
35  
36    protected TestMapEntrySetGenerator(
37        SampleElements<K> keys, SampleElements<V> values) {
38      this.keys = keys;
39      this.values = values;
40    }
41  
42    @Override
43    public SampleElements<Map.Entry<K, V>> samples() {
44      return SampleElements.mapEntries(keys, values);
45    }
46  
47    @Override
48    public Set<Map.Entry<K, V>> create(Object... elements) {
49      Map.Entry<K, V>[] entries = createArray(elements.length);
50      System.arraycopy(elements, 0, entries, 0, elements.length);
51      return createFromEntries(entries);
52    }
53  
54    public abstract Set<Map.Entry<K, V>> createFromEntries(
55        Map.Entry<K, V>[] entries);
56  
57    @Override
58    @SuppressWarnings("unchecked") // generic arrays make typesafety sad
59    public Map.Entry<K, V>[] createArray(int length) {
60      return new Map.Entry[length];
61    }
62  
63    /** Returns the original element list, unchanged. */
64    @Override
65    public List<Map.Entry<K, V>> order(List<Map.Entry<K, V>> insertionOrder) {
66      return insertionOrder;
67    }
68  }